package com.ruoyi.system.domain.util;

import com.ruoyi.system.domain.vo.TreeNode;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

/***
 * 说明
 * @author:ZZ
 * @date: 2024-12-03  17:47
 **/
public class TreeBuilder {

    public static TreeNode buildTree(List<TreeNode> nodes) {
        Map<String, TreeNode> map = new HashMap<>();
        for (TreeNode node : nodes) {
            map.put(node.getId(), node);
        }

        TreeNode root = null;
        for (TreeNode node : nodes) {
            TreeNode parent = map.get(node.getParentId());
            if (parent != null) {
                parent.getChildren().add(node);
            } else {
                root = node;
            }
        }
        return root;
    }
//
//    public static void main(String[] args) {
//        // 假设这是从数据库中查询到的数据
//        List<TreeNode> nodes = new ArrayList<>();
//        nodes.add(new TreeNode(1, 0));
//        nodes.add(new TreeNode(2, 1));
//        nodes.add(new TreeNode(3, 1));
//        nodes.add(new TreeNode(4, 2));
//        nodes.add(new TreeNode(5, 3));
//
//        TreeNode root = buildTree(nodes);
//        // 打印树形结构
//        printTree(root, 0);
//    }

    private static void printTree(TreeNode node, int level) {
        StringBuilder prefix = new StringBuilder();
        for (int i = 0; i < level; i++) {
            prefix.append("--");
        }
        System.out.println(prefix.toString() + node.getId());
        for (TreeNode child : node.getChildren()) {
            printTree(child, level + 1);
        }
    }

}
